package com.gaga.common.enums;

/**
 *
 */
public enum AiModelType {
    Qwen(1, "通义千问"),
    Doubao(2, "豆包"),
    ;

    private final Integer modelId;
    private final String desc;

    AiModelType(Integer modelId, String desc) {
        this.modelId = modelId;
        this.desc = desc;
    }

    public Integer getModelId() {
        return modelId;
    }

    public String getDesc() {
        return desc;
    }


    public static AiModelType getTypeByModelId(Long modelId) {
        AiModelType aiModelTypeResult = null;
        for (AiModelType aiModelType : AiModelType.values()) {
            if (aiModelType.getModelId().intValue() == modelId.intValue()) {
                aiModelTypeResult = aiModelType;
                break;
            }
        }
        return aiModelTypeResult;
    }
}
